package com.concurrency.example.proxy.cglibDemo;

import com.concurrency.example.proxy.TrainStation;
import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;

import java.lang.reflect.Method;

public class CglibFactory implements MethodInterceptor {

    /**
     * 获取代理对象
     * @return 代理对象
     */
    public TrainStation getProxy(){
        Enhancer enhancer = new Enhancer();
        //指定父类字节码
        enhancer.setSuperclass(TrainStation.class);
        //指定回调函数类，该类需实现 MethodInterceptor 接口
        enhancer.setCallback(this);
        //创建代理对象
        TrainStation trainStation = (TrainStation)enhancer.create();
        return trainStation;
    }

    /**
     * 回调函数
     * @param o 被代理的对象
     * @param method 代理方法
     * @param objects 代理参数
     * @param methodProxy 被代理的方法
     * @return 方法返回值
     * @throws Throwable
     */
    @Override
    public Object intercept(Object o, Method method, Object[] objects, MethodProxy methodProxy) throws Throwable {
        //增强处理
        System.out.println("通过Cglib代理，收取服务费");
        //执行被代理的方法
        return methodProxy.invokeSuper(o, objects);
    }
}
